from . import prompts

# 所有模型存放路径
MODEL_PATH = './Models/'
EMBEDDING_MODEL = 'bge-small-zh-v1.5'
entity_types = "[人员,技术,任务,组织,位置]"



CURRENT_KB = None

# 用于管理各类 API
class API_keys:
    # 搜索类api
    tavily =  "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"
    serper = "399a558e5d2ad3d34a240f260a52c0f71e2236b5"

    # 模型api
    # openai_api_url= "https://open.bigmodel.cn/api/paas/v4/"
    # openai_api_key = "cedac7d68c3cf5dff3593cd99b708dfa.Tm3L6Pl962gM0wgK"
    # openai_model_name = "glm-4"

    openai_api_url= 'http://10.11.129.202:7879/v1'
    openai_api_key = ""
    openai_model_name = "Qwen1.5"